package org.eoti.ai.genetic.gp;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;

/**
 * The MIT License
 * <p/>
 * Copyright (c) 2009 Malachi de AElfweald (malachid@gmail.com)
 * <p/>
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
 * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to the following conditions:
 * <p/>
 * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
 * Software.
 * <p/>
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
 * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
 * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */
public class Generation<DATA, FITNESS>
{
	private int id;
	private CopyOnWriteArrayList<Program<DATA, FITNESS>> population;
	private Criteria<DATA, FITNESS> criteria;

	public Generation(int id, Criteria<DATA, FITNESS> criteria)
	{
		this.id = id;
		this.criteria = criteria;
		this.population = new CopyOnWriteArrayList<Program<DATA, FITNESS>>();
	}

	public int getId(){return id;}

	public void randomize()
	{
		while(population.size() < criteria.getPopulationSize())
			population.add(createProgram());
	}

	protected Program<DATA, FITNESS> createProgram()
	{
		Program<DATA, FITNESS> program = new Program<DATA, FITNESS>(criteria);
		program.randomize( getCriteria() );
		return program;
	}

	public List<Program<DATA,FITNESS>> bestOfBreed()
	{
		ArrayList<Program<DATA,FITNESS>> list = new ArrayList<Program<DATA,FITNESS>>(population);
		Collections.sort(list, criteria.getFitness().getProgramComparator());
		return list;
	}

	public void add(Program<DATA,FITNESS> program)
	{
		if(size() >= criteria.getPopulationSize())
			return;

		if(!criteria.isAllowDuplicates() && population.contains( program ))
			return;

		population.add(program);
	}

	public int size(){return population.size();}

	public Criteria<DATA, FITNESS> getCriteria(){
		return criteria;
	}

	public void setCriteria( Criteria<DATA, FITNESS> criteria ){
		this.criteria = criteria;
	}
}
